package com.diy.sigmund.diycommon.util;

import java.util.ArrayList;
import java.util.List;
import java.util.function.BiConsumer;
import java.util.function.Supplier;
import org.springframework.beans.BeanUtils;

/**
 * 转换对象工具
 *
 * @author bugpool
 */
public class BeanConvertUtil extends BeanUtils {

    //    // 使用前
    //    public Vo findById(Integer id) {
    //        Vo vo = new Vo();
    //        Po po = dao.findById(id);
    //        copyProperties(po, vo);
    //        return vo;
    //    }
    //
    //    // 使用后
    //    public Vo findById(Integer id) {
    //        return BeanConvertUtils.converTo(dao.findById(id), Vo::new);
    //    }
    //
    //    // 使用后，通过lambda表达式特殊处理个别字段
    //    public Vo findById(Integer id) {
    //        return BeanConvertUtils.converTo(dao.findById(id), Vo::new,
    //                (s, t) -> t.setName(s.getName))
    // );
    //    }
    //
    //    // 使用前
    //    public List<Vo> findAll() {
    //        List<Vo> vos = new ArrayList();
    //        List<Po> pos = dao.findAll();
    //        for (Po po : Pos) {
    //            Vo vo = new Vo();
    //            BeanUtis.copyProperties(po, vo);
    //            vos.add(vo);
    //        }
    //        return vos;
    //    }
    //
    //    // 使用后
    //    public List<Vo> findAll() {
    //        return BeanConvertUtils.converToList(dao.findAll(), Vo::new)
    //    }
    //
    //    // 同样支持自定义lambda
    //    public List<Vo> findAll() {
    //        return BeanConvertUtils.converToList(dao.findAll(), Vo::new,
    //                (s, t) -> t.setName(s.getName))
    // )
    //    }


    public static <S, T> T convertTo(S source, Supplier<T> targetSupplier) {
        return convertTo(source, targetSupplier, null);
    }

    /**
     * 转换对象
     *
     * @param source         源对象
     * @param targetSupplier 目标对象供应方
     * @param biConsumer     回调方法
     * @param <S>            源对象类型
     * @param <T>            目标对象类型
     * @return 目标对象
     */
    public static <S, T> T convertTo(S source, Supplier<T> targetSupplier, BiConsumer<S, T> biConsumer) {
        if (null == source || null == targetSupplier) {
            return null;
        }

        T target = targetSupplier.get();
        copyProperties(source, target);
        if (biConsumer != null) {
            biConsumer.accept(source, target);
        }
        return target;
    }

    public static <S, T> List<T> convertListTo(List<S> sources, Supplier<T> targetSupplier) {
        return convertListTo(sources, targetSupplier, null);
    }

    /**
     * 转换对象
     *
     * @param sources        源对象list
     * @param targetSupplier 目标对象供应方
     * @param biConsumer     回调方法
     * @param <S>            源对象类型
     * @param <T>            目标对象类型
     * @return 目标对象list
     */
    public static <S, T> List<T> convertListTo(List<S> sources, Supplier<T> targetSupplier,
            BiConsumer<S, T> biConsumer) {
        if (null == sources || null == targetSupplier) {
            return null;
        }

        List<T> list = new ArrayList<>(sources.size());
        for (S source : sources) {
            T target = targetSupplier.get();
            copyProperties(source, target);
            if (biConsumer != null) {
                biConsumer.accept(source, target);
            }
            list.add(target);
        }
        return list;
    }
}
